package en.edu.zxj.mq.mqserver.core;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import en.edu.zxj.mq.common.ConsumerEnv;
import lombok.Data;

import java.util.*;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * Created with IntelliJ IDEA.
 * Description：消息队列,
 * 类名叫做 MSGQueue, ⽽不是 Queue, 是为了防⽌和标准库中的 Queue 混淆
 *
 * @author: zxj
 * @date: 2024-02-25
 * @time: 20:19:52
 */
@Data
public class MSGQueue {
    // 表示队列的身份标识.
    private String name;
    // 该消息队列是否需要持久化存储, true 表示需要持久化存储, false 表示不必持久化.
    private Boolean durable = false;


    // 以下为保留字段
    // exclusive 为 true, 表示这个队列只能被一个消费者使用(别人用不了), 如果为 false 则是大家都能使用
    private Boolean exclusive = false;
    // 该属性表示 如果当前交换机没有人用了, 就会自动删除
    private Boolean autoDelete = false;
    // arguments 表示的是创建交换机时指定的一些额外的参数选项, 待开发
    private Map<String, Object> arguments = new HashMap<>();


    // 必备字段
    Integer deleteFlag;
    Date createTime;
    Date updateTime;


    // 该队列表示被哪些消费者订阅
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();
    // 记录当前到哪个消费者, 方便采取轮询消费操作
    private AtomicInteger consumerSeq = new AtomicInteger(0);

    /*
     * @description: 添加一个新的订阅者, 订阅者的删除暂时不考虑
     **/
    public void addConsumerEnv(ConsumerEnv consumerEnv) {
        consumerEnvList.add(consumerEnv);
    }

    /**
     * @description: 以轮询的方式挑选一个订阅者, 用来处理当前的消费
     **/
    public ConsumerEnv chooseConsumer() {
        if (consumerEnvList.isEmpty()) {
            return null;
        }
        // 计算当前要取的元素的下标
        int index = consumerSeq.get() % consumerEnvList.size();
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }



    /**
     * @description: 这里的 getter 是提供给数据库使用的
     **/
    public String getArguments() {
        // 存储 arguments 字段在数据库中, 由于数据库没有哈希表结构, 需要使用 Json 转化成字符串才能存储
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        // 如果代码抛异常, 就返回空的 Json 对象
        return "{}";
    }

    /**
     * @description: 这里的 setter 是提供给数据库使用的
     **/
    public void setArguments(String arguments) {
        // arguments 在数据库总存储的是 String 类型, 从数据库获取的也是 String 类型
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(arguments, new TypeReference<Map<String, Object>>() {
            });
        } catch (JsonProcessingException e) {
            throw new RuntimeException(e);
        }
    }


    // 在这里针对 arguments, 再提供一组 getter setter , 用来去更方便的获取/设置这里的键值对.

    // 这一组在 java 代码内部使用 (比如测试的时候)
    public void setArguments(Map<String, Object> arguments) {
        this.arguments = arguments;
    }

    public Object getArguments(String key) {
        return arguments.get(key);
    }

    public void setArguments(String key,Object value) {
        arguments.put(key,value);
    }


}
